"""
将所有的标注帧都整合到一个文件夹中，为了方便yolov5检测
"""
import os
import shutil

# 源文件夹路径
source_folder = r'D:\file\postgrad\experiment\ava_dataset\videos_labelframes'
# 目标文件夹路径
target_folder = r'D:\file\postgrad\experiment\ava_dataset\merge_labelframes_all'

# 创建目标文件夹（如果不存在）
os.makedirs(target_folder, exist_ok=True)

# 遍历源文件夹及其所有子文件夹
for dirpath, dirnames, filenames in os.walk(source_folder):
    for filename in filenames:
        if filename.endswith('.jpg'):
            # 构造源文件路径
            src_file = os.path.join(dirpath, filename)
            # 构造目标文件路径
            dst_file = os.path.join(target_folder, filename)
            # 复制文件
            shutil.copy(src_file, dst_file)

print("所有 JPG 图片已整合到目标文件夹。")
